import torch
import template

template.print(torch.ones(3))